python -m torch.distributed.launch --nproc_per_node=2 --master_port 18101 --use_env main_imagenet.py --model vgg19_nn --pretrained --epochs 90 --batch-size 64 --lr 0.01 --wd 0.00004 --lr-step-size 1 --lr-gamma 0.98 --prune --cache-dataset --method group_norm --global_pruning --soft_keeping_ratio 0.5 --pretrained --target_flops 2.0 --output-dir run/imagenet_log/mob_ckpt 